# -*-coding:utf-8-*-
"""
使用 mplfinance 生成库生成图片
指南针指标通过python生成图片
"""

import pandas as pd
from dingding import dingding
from hander.HanderZnzView import HanderZnzView
from _cfg import base as cfg
import math
# 表名
index_0_table = 'index_0'
index_0a_table = 'index_0a'

hzv_obj = HanderZnzView()
hzv_obj._run() #清除文件夹
# 创建数据库连接
hzv_obj.create_sql_con(cfg=cfg)




def index0_and_index0b_create_img():
    dingding(title='指标图表生成信息',news=["指标信息图表生成","0号指标，0A指标，0B指标生成图表"])#发送钉钉消息
    print("#"*10)
    print('开始生成0号指数和0B指数的图表')

    # 0号指数和0B指数
    count_data= hzv_obj.get_data_count(table_name=index_0_table)
    count_num=count_data.iloc[0,0] #总数据
    num_=60 #每页数据
    page=1 #页码

    #总页数
    pages_=math.ceil(count_num/num_)
    pages_list=list(range(1,pages_+1))
    for page in pages_list:
        page_data = hzv_obj.get_data_from_sql(table_name=index_0_table, pages=page, num_=num_)
        hzv_obj._create_img_by_index0_pddata(page_data) #0号指数图表
        hzv_obj._create_img_by_index0b_pddata(page_data) #0B指数图表
    print('0号指数和0B指数的图表-生成成功！！')
    print("#"*10)


def index0a_create_img():

    print("#"*10)
    print('开始生成0A指数的图表')

    #0A指数
    count_data= hzv_obj.get_data_count(table_name=index_0a_table)
    count_num=count_data.iloc[0,0] #总数据
    num_=60 #每页数据
    page=1 #页码

    #总页数
    pages_=math.ceil(count_num/num_)
    pages_list=list(range(1,pages_+1))
    for page in pages_list:
        page_data = hzv_obj.get_data_from_sql(table_name=index_0a_table, pages=page, num_=num_)
        hzv_obj._create_img_by_index0a_pddata(page_data) #0A指数图表

    print('0A指数的图表-生成成功！！')
    print("#"*10)



if __name__=='__main__':

    #生成0A指数图表
    index0a_create_img()

    #生成0号和0B指数图表
    index0_and_index0b_create_img()